/* CoralCube | Massively Multiplayer Online Role-Playing Game(MMORPG) Emulator.
|* 
|* Copyright (C) 2005-2010 MaNGOS <http://getmangos.com/>
|*               2008-2010 TrinityCore <http://www.trinitycore.org/>
|*               2008-2010 CoralCube <http://www.CoralCube.com/>
|*
|* This program is free software. You can redistribute it and/or modify it under the terms of 
|* the GNU General Public License as published by the Free Software Foundation, either 
|* version 2 of the License.
|*
|* This program is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY, 
|* without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. 
|* See the GNU General Public License for more details.
|*
|* You should have received a copy of the GNU General Public License along with this program. 
|* If not, see <http://www.gnu.org/licenses/>.
|*/

#ifndef QUERYRESULT_H
#define QUERYRESULT_H

#include <ace/Refcounted_Auto_Ptr.h>
#include <ace/Thread_Mutex.h>

#include "Field.h"
#include "Log.h"

#ifdef _WIN32
#include <winsock2.h>
#endif
#include <mysql.h>

class ResultSet
{
public:
    ResultSet(MYSQL_RES *result, MYSQL_FIELD *fields, uint64 rowCount, uint32 fieldCount);
    ~ResultSet();

    bool GetFieldNames();
    bool NextRow();
    uint64 GetRowCount() const { return m_rowCount; }
    uint32 GetFieldCount() const { return m_fieldCount; }

    Field *Fetch() const { return m_currentRow; }
    const Field & operator [] (uint32 index) const
    {
        ASSERT(index < m_fieldCount);
        return m_currentRow[index];
    }

protected:
    Field *m_currentRow;
    uint64 m_rowCount;
    uint32 m_fieldCount;

private:
    void CleanUp();
    MYSQL_RES *m_result;
    MYSQL_FIELD *m_fields;
};

typedef ACE_Refcounted_Auto_Ptr<ResultSet, ACE_Thread_Mutex> QueryResult;

class PreparedResultSet
{
public:
    PreparedResultSet(MYSQL_STMT* stmt, MYSQL_RES *result, uint64 rowCount, uint32 fieldCount);
    ~PreparedResultSet();

    bool NextRow();
    uint64 GetRowCount() const { return m_rowCount; }
    uint32 GetFieldCount() const { return m_fieldCount; }

    Field* Fetch() const
    {
        ASSERT(m_rowPosition < m_rowCount);
        return m_rows[uint32(m_rowPosition)];
    }

    const Field & operator [] (uint32 index) const
    {
        ASSERT(m_rowPosition < m_rowCount);
        ASSERT(index < m_fieldCount);
        return m_rows[uint32(m_rowPosition)][index];
    }

protected:
    uint64 m_rowCount;
    uint64 m_rowPosition;
    std::vector<Field*> m_rows;
    uint32 m_fieldCount;

private:
    MYSQL_BIND* m_rBind;
    MYSQL_STMT* m_stmt;
    MYSQL_RES* m_res;

    my_bool* m_isNull;
    unsigned long* m_length;

    void FreeBindBuffer();
    void CleanUp();
    bool _NextRow();
};

typedef ACE_Refcounted_Auto_Ptr<PreparedResultSet, ACE_Thread_Mutex> PreparedQueryResult;

typedef std::vector<std::string> QueryFieldNames;

class QueryNamedResult
{
public:
    explicit QueryNamedResult(ResultSet* query, QueryFieldNames const& names) : mQuery(query), mFieldNames(names) { }
    ~QueryNamedResult() { delete mQuery; }

    // compatible interface with ResultSet
    bool NextRow() { return mQuery->NextRow(); }
    Field *Fetch() const { return mQuery->Fetch(); }
    uint32 GetFieldCount() const { return mQuery->GetFieldCount(); }
    uint64 GetRowCount() const { return mQuery->GetRowCount(); }
    Field const& operator[] (int index) const { return (*mQuery)[index]; }

    // named access
    Field const& operator[] (const std::string &name) const { return mQuery->Fetch()[GetField_idx(name)]; }
    QueryFieldNames const& GetFieldNames() const { return mFieldNames; }

    uint32 GetField_idx(const std::string &name) const
    {
        for (size_t idx = 0; idx < mFieldNames.size(); ++idx)
        {
            if (mFieldNames[idx] == name)
                return idx;
        }
        ASSERT(false && "unknown field name");
        return uint32(-1);
    }

protected:
    ResultSet *mQuery;
    QueryFieldNames mFieldNames;
};
#endif
